import torch
print(torch.__version__)          # 查看 PyTorch 版本
print(torch.cuda.is_available())  # 应返回 True
print(torch.version.cuda)         # 应返回 12.1（兼容 12.9）
print(torch.backends.cudnn.version())  # 应返回 cuDNN 版本